op {
  name: "RetrieveTPUEmbeddingADAMParameters"
  output_arg {
    name: "parameters"
    type: DT_FLOAT
  }
  output_arg {
    name: "momenta"
    type: DT_FLOAT
  }
  output_arg {
    name: "velocities"
    type: DT_FLOAT
  }
  attr {
    name: "table_id"
    type: "int"
    default_value {
      i: -1
    }
    has_minimum: true
    minimum: -1
  }
  attr {
    name: "table_name"
    type: "string"
    default_value {
      s: ""
    }
  }
  attr {
    name: "num_shards"
    type: "int"
  }
  attr {
    name: "shard_id"
    type: "int"
  }
  is_stateful: true
}
op {
  name: "RetrieveTPUEmbeddingADAMParameters"
  output_arg {
    name: "parameters"
    type: DT_FLOAT
  }
  output_arg {
    name: "momenta"
    type: DT_FLOAT
  }
  output_arg {
    name: "velocities"
    type: DT_FLOAT
  }
  attr {
    name: "table_id"
    type: "int"
    default_value {
      i: -1
    }
    has_minimum: true
    minimum: -1
  }
  attr {
    name: "table_name"
    type: "string"
    default_value {
      s: ""
    }
  }
  attr {
    name: "num_shards"
    type: "int"
  }
  attr {
    name: "shard_id"
    type: "int"
  }
  attr {
    name: "config"
    type: "string"
    default_value {
      s: ""
    }
  }
  is_stateful: true
}
op {
  name: "RetrieveTPUEmbeddingADAMParameters"
  output_arg {
    name: "parameters"
    type: DT_FLOAT
  }
  output_arg {
    name: "momenta"
    type: DT_FLOAT
  }
  output_arg {
    name: "velocities"
    type: DT_FLOAT
  }
  attr {
    name: "table_id"
    type: "int"
    default_value {
      i: -1
    }
  }
  attr {
    name: "table_name"
    type: "string"
    default_value {
      s: ""
    }
  }
  attr {
    name: "num_shards"
    type: "int"
  }
  attr {
    name: "shard_id"
    type: "int"
  }
  attr {
    name: "config"
    type: "string"
    default_value {
      s: ""
    }
  }
  is_stateful: true
}
